device = None

def get_torch_device():
    return device
